import modules.range
import modules.firepower

class ModuleFactory():
    """Returns the correct class for the given ship"""
    
    def __init__(self):
        pass

    def create_module(self, type):
        #help(modules.range.Range())
        if type == 'EnergyStorage':
            class_tmp = EnergyStore()
        elif type == 'MassStorage':
            class_tmp = MassStore()
        elif type == 'MultiStorage':
            class_tmp = MultiStorage()
        elif type == 'EnergyReduction':
            class_tmp = EnergyReduction()
        elif type == 'MassReduction':
            class_tmp = MassReduction()
        elif type == 'MultiReduction':
            class_tmp = MultiReduction()
        elif type == 'FireRate':
            class_tmp = FireRate()
        elif type == 'FirePower':
            class_tmp = modules.firepower.FirePower()
        elif type == 'Range':
            class_tmp = modules.range.Range()
        else:
            pass                                                
        return class_tmp
    
        